import numpy as np

# 创建一个二维数组
arr_2d = np.array([[5, 2, 6],[3, 4, 1]])

# 对每一列进行排序
sorted_arr_2d = np.sort(arr_2d, axis=-1)

print(sorted_arr_2d)